
from func import user
from Acc import account

def strat_menu():
    print('{}{}{}'.format('*' * 15, '欢迎来到WoniuATM', '*' * 15))
    print('{}{}{}'.format('*' * 15, ' 请选择操作菜单 ', '*' * 15))
    print('{}{}{}'.format('*' * 13, '1.注册 2.登录 3.退卡', '*' * 13))
    choose = input('你想干啥')
    if choose == '1':
        user().register()
    elif choose == '2':
        user().login()
    elif choose == '3':
        exit('欢迎下次再来')
    else:
        print('没有这个选项')
        strat_menu()


def main_menu():
    print('{}{}{}'.format('*' * 21, '恭喜登录蜗牛银行', '*' * 21))
    print('{}{}{}'.format('*' * 21, ' 请选择操作菜单 ', '*' * 21))
    print('{}{}{}'.format('*' * 5, '1.查询 2.取款 3.存款 4.转账 5.流水 6.返回 7.退卡', '*' * 5))
    kk = input('你要干啥')
    if kk == '1':
        account().select()
    elif kk == '2':
        account().withdraw()
    elif kk == '3':
        account().deposite()
    elif kk == '4':
        account().transfer()
    elif kk == '5':
        account().statement()
    elif kk == '6':
        strat_menu()
    elif kk == '7':
        exit('欢迎下次再来')
    else:
        print('没有这个选项')
        main_menu()


if __name__ == '__main__':
    strat_menu()
if __name__=='__main__':
    strat_menu() 
